#!/bin/bash
###---------cann8.2.rc1------------
docker build . \
    -t cann8.2.rc1-py310-ubuntu20.04-arm64:0831 --no-cache -f Dockerfile.cann

###--------pytorch2.6.0-------------
docker build . \
    --build-arg BASE="cann8.2.rc1-py310-ubuntu20.04-arm64:0831" \
    -t pytorch2.6.0-cann8.2.rc1-py310-ubuntu20.04-arm64:0831 --no-cache -f  Dockerfile.pta

###--------mindspeed-llm2.1.0-------------
docker build . \
    --build-arg BASE="pytorch2.6.0-cann8.2.rc1-py310-ubuntu20.04-arm64:0831" \
    -t mindspeed-llm2.1.0-pytorch2.6.0-cann8.2.rc1-py310-ubuntu20.04-arm64:0831 --no-cache -f  Dockerfile.mindspeed_llm

###--------mindspeed-llm master+2.1.0-------------
docker build . \
    --build-arg BASE="pytorch2.6.0-cann8.2.rc1-py310-ubuntu20.04-arm64:0831" \
    -t mindspeed-llm_master_2.1.0-pytorch2.6.0-cann8.2.rc1-py310-ubuntu20.04-arm64:0831 --no-cache -f Dockerfile.mindspeed_llm_master_210
###--------mindspeed-rl-mm 2.1.0-------------
docker build . \
    --build-arg BASE="pytorch2.5.1-cann8.2.rc1-py310-ubuntu20.04-arm64:0831" \
    -t mindspeed-rl-mm_2.1.0-pytorch2.5.1-cann8.2.rc1-py310-ubuntu20.04-arm64:0831 -f Dockerfile.mindspeed_rl_mm_210
###--------mindspeed-rl-llm 2.1.0-------------
docker build . \
    --build-arg BASE="pytorch2.5.1-cann8.2.rc1-py310-ubuntu20.04-arm64:0831" \
    -t mindspeed-rl-llm_2.1.0-pytorch2.5.1-cann8.2.rc1-py310-ubuntu20.04-arm64:0831 -f Dockerfile.mindspeed_rl_llm_210
